//
//  HMPictureSelectorController.swift
//  新浪微博
//
//  Created by 饶鹏伟 on 16/4/19.
//  Copyright © 2016年 饶鹏伟. All rights reserved.
//

//图片选择控制器


import UIKit
import SnapKit
import HMImagePicker//图片选择的第三方框架
//view上每个cell的间距及两边的距离
private let cellMargin: CGFloat=3
//每行显示几个cell
private let colCount: CGFloat=4
//最大的可选图片张数
private let maxImageCount=9
//可重用cell标示
private let PicturSelectorCellId = "PicturSelectorCellId"


class HMPictureSelectorController: UICollectionViewController {
    

    //定义一个数组 把选择的图片放到数组中
    lazy var imageList: [UIImage]=[UIImage]()
    //记录图片选择器选择了哪些照片
    private var selectedAsset: [PHAsset]?
    
    override func viewDidLoad() {
        super.viewDidLoad()

        //注册cell
        self.collectionView!.registerClass(PictureSelectorCell.self, forCellWithReuseIdentifier: PicturSelectorCellId)
    }

   
    init(){
        
       //设置流水布局
        
        //每个cell的宽度
        let itemW=(ScreenWidth-(colCount+1)*cellMargin)/colCount
        //实例化布局对象
        let layOut=UICollectionViewFlowLayout()
        
        //设置每个cell之间的间距
        layOut.minimumInteritemSpacing=cellMargin
        layOut.minimumLineSpacing=cellMargin
        
        //设置布局对象的范围
        layOut.sectionInset=UIEdgeInsets(top: cellMargin, left: cellMargin, bottom: 0, right: cellMargin)
        //设置每个cell的大小
        layOut.itemSize=CGSize(width: itemW, height: itemW)
        
        super.init(collectionViewLayout: layOut)
        
        collectionView?.backgroundColor=UIColor.whiteColor()
        
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    
    
   
}


//实现数据源方法
extension HMPictureSelectorController{
    
    //每组返回多少个cell
    override func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        
        //始终显示比选择了图片多一个按钮
        let detla = imageList.count >= maxImageCount ? 0 : 1
        
       return imageList.count + detla
    }
    
    
    //返回每个cell
    
    override func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
        //获取重用cell
        let cell=collectionView.dequeueReusableCellWithReuseIdentifier(PicturSelectorCellId, forIndexPath: indexPath) as! PictureSelectorCell
        
        //设置cell颜色
        cell.backgroundColor=randomColor()
        
        //指定代理
        cell.selectorDelegate=self
        
        print(indexPath.item)
        
        //设置每个cell的图片
        if indexPath.item == imageList.count {
            //程序走到这个地方  如果去数组根据索引取值 就会造成数组索引越界
            cell.image = nil
        } else {
            cell.image = imageList[indexPath.item]
        }
        
        return cell
    }
    
    
}


//遵守代理协议 实现协议方法
extension HMPictureSelectorController: PictureSelectorCellDelegate{
    
    //实现协议的选择图片方法
    func userWillChosePicture() {
        
        //如果允许图片选择器 访问相册
        if UIImagePickerController.isSourceTypeAvailable(.PhotoLibrary){
            
            //使用框架 选择图片 实例化一个图片选择器对象 参数是记录了选择了哪些照片
            let picker=HMImagePickerController(selectedAssets: selectedAsset)
            //设置代理
            picker.pickerDelegate=self
            
            //设置目标图片尺寸
            picker.targetSize=CGSizeMake(600, 600)
            //设置最多选择 的 图片个数 默认9张
            picker.maxPickerCount=maxImageCount

            //跳转到图片选择控制器
            presentViewController(picker, animated: true, completion: nil)
        }
        
        
        
    }
    
    //实现协议的删除图片的方法
    func userWillDeletePicture(cell: PictureSelectorCell) {
        
        // 获取点击cell、的indexpath
        if let indexPath = collectionView?.indexPathForCell(cell){
            
           //在数组中移除对应 indexPath.item 对应的元素
            imageList.removeAtIndex(indexPath.item)
            
           //删除上一次选择图片的数据源
            self.selectedAsset?.removeAtIndex(indexPath.item)
            
           //刷新数据
            collectionView?.reloadData()
        }
        
        
    }
    
}




//遵守图片选择器的协议 并实现方法

extension HMPictureSelectorController: HMImagePickerControllerDelegate{
    
    
    //实现协议方法  一旦实现了协议方法之后 图片选择器的消失 就交给程序员来管理
    func imagePickerController(picker: HMImagePickerController, didFinishSelectedImages images: [UIImage], selectedAssets: [PHAsset]?) {
         //把选择的图片储存起来
        self.imageList=images
        //记录选中资源集合，方便再次选择照片定位
        self.selectedAsset=selectedAssets
        //修改了数据源之后 就应该刷新数据
        self.collectionView?.reloadData()
        //选择图片完成后 手动退出图片选择控制器
        dismissViewControllerAnimated(true, completion: nil)
    }
    

}



//自定义cell的一个协议 让控制器遵守 实现方法
@objc protocol PictureSelectorCellDelegate: NSObjectProtocol{
    //用户将要选择图片
   optional func userWillChosePicture()
    
    //用户将要删除图片
    optional func userWillDeletePicture(cell: PictureSelectorCell)
}




//自定义图片选择的每个cell

class PictureSelectorCell: UICollectionViewCell {
    
    
    //每个cell按钮上的图片属性
    var image: UIImage?{
        
        didSet{
            
           //如果有图片 显示删除按钮 如果没图片 隐藏删除按钮
            deleteBtn.hidden=image==nil
            
            //设置按钮图片
            addBtn.setImage(image, forState: .Normal)
            
        }
        
    }
    
    
    // 定义代理对象属性 弱引用
    weak var selectorDelegate: PictureSelectorCellDelegate?
    
    override init(frame: CGRect) {
        
        super.init(frame: frame)
        
        //设置UI界面
        setupUI()
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    //设置UI界面
    private func setupUI(){
        
        //添加cell上面的子视图按钮
        contentView.addSubview(addBtn)
        contentView.addSubview(deleteBtn)
        
        //设置约束
        addBtn.snp_makeConstraints { (make) -> Void in
            
            make.edges.equalTo(contentView.snp_edges)
        }
        
        deleteBtn.snp_makeConstraints { (make) -> Void in
            
            make.top.equalTo(contentView.snp_top)
            make.right.equalTo(contentView.snp_right)
        }
        
        //给按钮添加点击事件
        addBtn.addTarget(self, action: "addBtnDidClick", forControlEvents: .TouchUpInside)
        deleteBtn.addTarget(self, action: "deleteBtnDidClick", forControlEvents: .TouchUpInside)
        
        
    }
    
    
    //实现按钮的点击事件
    //添加图片的按钮
    @objc private func addBtnDidClick(){
        
        print("添加图片")
       
       //判断cell是否有图 如果有图 不需要添加图片了
        if image != nil{
            
            print("已经有图片了")
            
            return
        }
        
        //如果没有图片 调用代理对象的添加图片的方法
         selectorDelegate?.userWillChosePicture?()
    }
    
    
    //删除点击按钮的图片的按钮
    @objc private func deleteBtnDidClick(){
        
        //点击删除按钮 调用代理对象方法 删除当前点击的cell 的图片
        selectorDelegate?.userWillDeletePicture?(self)
        
        print("删除图片")
    }
    
    
    
   //懒加载 添加图片的按钮 在每个cell上面添加一个按钮
    private lazy var addBtn: UIButton={
        //实例化按钮
        let btn=UIButton()
        //设置按钮图片的填充方式
        btn.imageView?.contentMode = .ScaleAspectFill
        
        //设置按钮的背景图片
        btn.setBackgroundImage(UIImage(named: "compose_pic_add"), forState: .Normal)
        btn.setBackgroundImage(UIImage(named: "compose_pic_add_highlighted"), forState: .Highlighted)
        
        return btn
    }()
    
    
    //懒加载 删除图片的按钮 在每个cell上面
    private lazy var deleteBtn: UIButton={
        
        let btn=UIButton()
        
        btn.setBackgroundImage(UIImage(named: "compose_photo_close"), forState: .Normal)

        
        return btn

    }()
    
}

